SIBLE_DEVICES=5,6 colossalai run --nproc_per_node 2 train_multi_gpu_colossalai.py
